numpy applynumpy vectorize